tensorflow input pipeline performance